package com.sicau.config;

import com.sicau.rag.TokenAndTrunkCountBatchingStrategy;
import org.springframework.ai.embedding.BatchingStrategy;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

/**
 * 自定义向量模型配置
 */
@Configuration
public class EmbeddingStrategyConfig {
    @Bean
    public BatchingStrategy customTokenCountBatchingStrategy() {
        return new TokenAndTrunkCountBatchingStrategy();
    }
}